package com.tanker.common.utils;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Date;
import java.util.Random;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;

import com.aliyun.oss.OSSClient;
import com.aliyun.oss.model.ObjectMetadata;
import com.aliyun.oss.model.PutObjectResult;

/**
 * @author xqxue
 * @time 2017年9月19日 下午2:39:09
 */
public class OSSClientUtil {
	private Logger log = LoggerFactory.getLogger(this.getClass());
	// endpoint以杭州为例，其它region请按实际情况填写
	private String endpoint = "http://oss-cn-hangzhou.aliyuncs.com";
	// accessKey
	private String accessKeyId = "XakvqZLxRN1DR9Iy";
	private String accessKeySecret = "6GNGlPXmKe3uKC1GQJs5avnRHuOeE2";
	// 空间
	private String bucketName;// = "at-images-test";
	// 文件存储目录
	private String filedir = "tanker/";

	private OSSClient ossClient;

	public OSSClientUtil() {
		ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
	}

	/**
	 * 销毁
	 */
	public void destory() {
		// ossClient.shutdown();
	}

	/**
	 * 上传图片
	 * @param url
	 */
	/*public void uploadImg2Oss(String url) {
		File fileOnServer = new File(url);
		FileInputStream fin;
		try {
			fin = new FileInputStream(fileOnServer);
			String[] split = url.split("/");
			this.uploadFile2OSS(fin, split[split.length - 1]);
		} catch (FileNotFoundException e) {
			log.error("上传图片失败 ");
			e.printStackTrace();
		}
	}*/

	public String uploadImg2Oss(MultipartFile file, int memNo) {
		String originalFilename = file.getOriginalFilename();
		String substring = originalFilename.substring(
				originalFilename.lastIndexOf(".")).toLowerCase();
		Random random = new Random();
		String name = random.nextInt(10000) + System.currentTimeMillis()
				+ substring;
		InputStream inputStream = null;
		String path = memNo + "/" + name;
		try {
			inputStream = file.getInputStream();
			this.uploadFile2OSS(inputStream, path, memNo);
			return path;
		} catch (Exception e) {
			log.error("上传图片失败 ");
			e.printStackTrace();
		} finally {
			try {
				if(inputStream!=null) 
					inputStream.close();
			} catch(Exception e) {
				e.printStackTrace();
			}
		}
		return null;
	}

	/**
	 * 获得图片路径
	 * @param fileUrl
	 * @return
	 */
	public String getImgUrl(String fileUrl) {
		if (!StringUtils.isEmpty(fileUrl)) {
			String[] split = fileUrl.split("/");
			return this.getUrl(this.filedir + split[split.length - 1]);
		}
		return "";
	}

	/**
	 * 上传到OSS服务器 如果同名文件会覆盖服务器上的
	 * @param instream 文件流
	 * @param fileName  文件名称 包括后缀名
	 * @return 出错返回"" ,唯一MD5数字签名
	 */
	public String uploadFile2OSS(InputStream instream, String fileName, int memNo) {
		String ret = "";
		try {
			// 创建上传Object的Metadata
			ObjectMetadata objectMetadata = new ObjectMetadata();
			objectMetadata.setContentType("image/jpg");// 在metadata中标记文件类型
			// objectMetadata.setContentLength(instream.available());
			// objectMetadata.setCacheControl("no-cache");
			// objectMetadata.setHeader("Pragma", "no-cache");
			// objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf("."))));
			// objectMetadata.setContentDisposition("inline;filename=" +
			// fileName);
			// 上传文件
			PutObjectResult putResult = ossClient.putObject(bucketName, filedir + fileName, instream, objectMetadata);
			ret = putResult.getETag();
		} catch (Exception e) {
			log.error(e.getMessage(), e);
		} finally {
			try {
				if (instream != null) {
					instream.close();
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return ret;
	}

	/**
	 * Description: 判断OSS服务文件上传时文件的contentType
	 * @param FilenameExtension  文件后缀
	 */
	public static String getContentType(String FilenameExtension) {
		if (FilenameExtension.equalsIgnoreCase("bmp")) {
			return "image/bmp";
		}
		if (FilenameExtension.equalsIgnoreCase("gif")) {
			return "image/gif";
		}
		if (FilenameExtension.equalsIgnoreCase("jpeg")
				|| FilenameExtension.equalsIgnoreCase("jpg")
				|| FilenameExtension.equalsIgnoreCase("png")) {
			return "image/jpeg";
		}
		if (FilenameExtension.equalsIgnoreCase("html")) {
			return "text/html";
		}
		if (FilenameExtension.equalsIgnoreCase("txt")) {
			return "text/plain";
		}
		if (FilenameExtension.equalsIgnoreCase("vsd")) {
			return "application/vnd.visio";
		}
		if (FilenameExtension.equalsIgnoreCase("pptx")
				|| FilenameExtension.equalsIgnoreCase("ppt")) {
			return "application/vnd.ms-powerpoint";
		}
		if (FilenameExtension.equalsIgnoreCase("docx")
				|| FilenameExtension.equalsIgnoreCase("doc")) {
			return "application/msword";
		}
		if (FilenameExtension.equalsIgnoreCase("xml")) {
			return "text/xml";
		}
		return "image/jpeg";
	}

	/**
	 * 获得url链接
	 * @param key
	 * @return
	 */
	public String getUrl(String key) {
		String url = "";
		if (!StringUtils.isEmpty(key)) {
			// 设置URL过期时间为10年 3600l* 1000*24*365*10
			Date expiration = new Date(new Date().getTime() + 3600l * 1000 * 24
					* 365 * 10);
			// 生成URL
			String path = this.filedir + key;
			URL u = ossClient.generatePresignedUrl(bucketName, path, expiration);
			if (u != null) {
				url = u.getAuthority() + u.getPath();
			}
		}
		
		return url;
	}

	public String getEndpoint() {
		return endpoint;
	}

	public void setEndpoint(String endpoint) {
		this.endpoint = endpoint;
	}

	public String getAccessKeyId() {
		return accessKeyId;
	}

	public void setAccessKeyId(String accessKeyId) {
		this.accessKeyId = accessKeyId;
	}

	public String getAccessKeySecret() {
		return accessKeySecret;
	}

	public void setAccessKeySecret(String accessKeySecret) {
		this.accessKeySecret = accessKeySecret;
	}

	public String getBucketName() {
		return bucketName;
	}

	public void setBucketName(String bucketName) {
		this.bucketName = bucketName;
	}
	
}
